import torch
import torch_npu
import sys

sys.path.append("./build")
import ascend_ops

a = torch.tensor([1, 2, 3], dtype=torch.float32)
b:torch.Tensor = a.npu()
for d in [
    torch.half,
    torch.float16,
    torch.float,
    torch.float32,
    torch.float64,
    torch.int8,
    torch.uint8,
    torch.int16,
    torch.int,
    torch.int64,
    torch.bool,
]:
    b = b.to(dtype=d)
    ascend_ops.AclTensor(b)
stream = ascend_ops.getCurrentStream()
print(stream, type(stream))
